diff --git a/arch/riscv/lib/cache.c b/arch/riscv/lib/cache.c
index 5437a12..b1d42bc 100644
--- a/arch/riscv/lib/cache.c
+++ b/arch/riscv/lib/cache.c
@@ -5,6 +5,7 @@
  */
 
 #include <common.h>
+#include <cpu_func.h>
 
 void invalidate_icache_all(void)
 {